import erniebot


# 初始化 Erniebot
def initialize_erniebot():
    erniebot.api_type = 'aistudio'
    erniebot.access_token = '721efe09951433260c4ab297c2830cffc82e2d77'


def polish(user_query):
    initialize_erniebot()

    question = f"""
    你是一位资深的语文老师，需要对一段话中进行优化和润色，让这段话整体更加饱满。请按以下要求执行：

    针对一段话中的语句进行优化调整。
    1.理解语境，整句优化，对原文整段句子进行扩写、改写、润色，不要针对【个别词汇】优化。
    2.不要纠正原文中的错别字，需要改写整段句子。
    3.使用更专业的词汇和修辞手法进行润色。
    4.输出此段话优化后的所有文本。
    5.严格按照[示例]中的格式输出。
    [示例] 原始段落输入： 生活并不总是顺其自然，我们总是在困难面前找寻出路。

    优化后的输出： 生活从不会简单地顺从命运的安排，我们总在逆境中寻求突破的道路。

    [原段落]
    {user_query}
    """

    response = erniebot.ChatCompletion.create(
        model='ernie-turbo',
        messages=[
            {'role': 'user', 'content': question},
        ],
        temperature=0.2
    )

    return response.get_result()

# 调用方式示例（在其他文件中，首先要导入这个函数）
# user_question = input()
# result = polish(user_question)
# print(result)
